[release 2.11] Update to torch 2.11#34644
Conversation
There was a problem hiding this comment.
Code Review
This pull request updates the PyTorch dependency from version 2.10.0 to 2.11.0, along with its companion libraries torchvision (to 0.26.0) and torchaudio (to 2.11.0). The changes are applied consistently across various configuration files, including CMakeLists.txt, Dockerfiles, and requirements files for different platforms (CPU, CUDA, ROCm, XPU). The update appears to be a straightforward and consistent version bump. My review did not identify any critical or high-severity issues in the submitted changes.
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
6e3df89 to
7b6c567
Compare
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
9 similar comments
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
|
This pull request has merge conflicts that must be resolved before it can be |
2b87067 to
0b99aaa
Compare
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit>=4.5.1
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
| # glibc version is baked into the distro, and binaries built with one glibc | ||
| # version are not backwards compatible with OSes that use an earlier version. | ||
| ARG BUILD_BASE_IMAGE=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04 | ||
| ARG BUILD_BASE_IMAGE=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04 |
There was a problem hiding this comment.
Updating the build base image Ubuntu version will likely affect the glibc support, as noted in the comment above. Are there no CUDA 13 images available for 20.04? Cc @tlrmchlsmth
There was a problem hiding this comment.
Hi @mgoin no unfortunately they start with 20.04: https://hub.docker.com/r/nvidia/cuda/tags?name=13.0.0-base-ubuntu
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit>=4.5.1
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit>=4.5.1
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
Signed-off-by: atalman <atalman@fb.com>
Signed-off-by: atalman <atalman@fb.com>
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit>=4.5.1
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
|
Kernels Helion Test fixed by: #38062 Pre-Commit looks like existing issue: |
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit>=4.5.1
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
|
This pull request has merge conflicts that must be resolved before it can be |
Signed-off-by: Andrey Talman <atalman@fb.com>
|
Hi @atalman, the pre-commit checks have failed. Please run: uv pip install pre-commit>=4.5.1
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
|
Documentation preview: https://vllm--34644.org.readthedocs.build/en/34644/ |
|
I enabled some tests for ROCm as well. If possible before next rebase (if there is any) let's wait for those tests to complete. I will check the results here again in about 3 hrs if that is acceptable. |
AndreasKaratzas
left a comment
There was a problem hiding this comment.
Only weird part I saw was compared to our nightly: https://buildkite.com/vllm/amd-ci/builds/7368/steps/canvas?sid=019d67ff-82fe-4266-bf39-09f5be890841&tab=output#019d67ff-83f9-433b-8a9e-aa451f43d925/L2538
But this should not be related to AMD CI
| # Note: also needed to run plamo2 model in vLLM | ||
| - uv pip install --system --no-build-isolation 'git+https://github.com/state-spaces/mamba@v2.3.0' | ||
| - uv pip install --system --no-build-isolation 'git+https://github.com/Dao-AILab/causal-conv1d@v1.5.2' | ||
| - uv pip install --system --no-build-isolation 'git+https://github.com/Dao-AILab/causal-conv1d@v1.6.0' |
There was a problem hiding this comment.
Can we also change this version in test-amd.yaml? Btw was there a problem with 1.5.2?
| # we can only upgrade after this is resolved | ||
| # TODO(jerryzh168): resolve the above comment | ||
| - uv pip install --system torchao==0.14.1 --index-url https://download.pytorch.org/whl/cu129 | ||
| - uv pip install --system torchao==0.17.0 --index-url https://download.pytorch.org/whl/cu130 |
There was a problem hiding this comment.
Torchao seems to be consistent in its failures between this version and our current nightly:
- https://buildkite.com/vllm/amd-ci/builds/7364/steps/canvas?sid=019d6686-f3b8-4cf7-b5d8-f295e9246ce9&tab=output#019d6686-f4ac-4f0e-a2fb-2678f5cbf103/L14384
- https://buildkite.com/vllm/amd-ci/builds/7364/steps/canvas?sid=019d6686-f37f-4297-b7bf-41c0d80c2cf5&tab=output#019d6686-f47a-4044-aef8-ba982680e9e0/L6902
Therefore, there is no obvious problem here for now.
|
thanks for your kindly upgrade for xpu. while xpu have its low level dependency (oneapi, just like nvidia platform have cuda dependency as well). fortunately, 2.10 and 2.11 are share same base dependency version, so everything works fine. Intel will handle further torch-xpu version upgrade to avoid potential break. |
Update to Torch 2.11